记录一下windows下openCV(tbb+cuda)的编译安装

您所在的位置:网站首页 opencv 架构 记录一下windows下openCV(tbb+cuda)的编译安装

记录一下windows下openCV(tbb+cuda)的编译安装

2023-04-14 19:52| 来源: 网络整理| 查看: 265

在整 openCV 的时候为了玩到 cuda 和 tbb 编译整到麻,编译十万年,报错十万年,所以简单记录一下。

此处使用 CMake + VS 编译。

1. 源码

下载opencv源码和opencv_contrib源码

此处需要两者的版本完全一致,这里使用如下代码,其中 X.X.X 填写需要的版本

git clone https://github.com/opencvlopencv.git cd opencv git checkout X.X.X cd .. git clone https://github.com/opencv_contrib.git cd opencv_contrib git checkout X.X.X 2. oneAPI(tbb)

为了获得 tbb ,直接下载 oneAPI 并安装,完全安装会很大,但是只安装 tbb 即可。

3. VTK

下载VTK源码(source项)并用CMake编译成VS项目,随后打开项目,修改为生成 Release x64 ,随后生成 ALL_BUILD 项,再生成 INSTALL 项即安装完成。

4. cuda和cuDNN

进入英伟达官网下载cuda toolkit,下载安装对应显卡的版本,需要注意编译 openCV 的时候有版本的要求,因此应当适当的选大一点(此文章采用的版本为 ≥11.6 ),并设置全局变量(似乎安装的时候有自动设置的选项?如果没设置上就搜一下设置哪些选项就是了)。

同理下载cuDNN并解压到对应文件夹。

5. CMake Configure操作

在 CMake 里选择 openCV 的源码文件夹和 build 文件夹,Configure 一遍,搜索如下项

勾选WITH_TBB勾选WITH_OPEN_MP勾选ENABLE_FAST_MATH勾选BUILD_OPENCV_WORLD搜索cuda,全勾上搜索EXTRA_MODULES,填上 opencv_contrib/modules 的路径(windows下的斜杠需要换个方向)如果不需要 python 、 java 等可以搜索这几项后把勾去了

接着 Configure 一遍,找到 CUDA_ARCH_BIN 这一项。这一项是编译的代码对应的 GPU 架构,在英伟达官网找到自己的 GPU 的对应,将除了这一对应的其它版本都删了。如果不删除这一项编译时间会长很多倍

在 configure 的过程中有可能会产生下载的东西错误(比如 raw.githubusercontent 无法连接上)。此时需要科学上网或者从网上找到对应的资源下载下来,并放在对应文件夹(具体文件夹查询 build/CMakeDownloadLog.txt)。在没有任何错误之后后再进行下一步。

6.build

用 CMake 进行 Generate 后进入VS项目。使用 Debug 和 Release 分别进行以下操作:生成 ALL_BUILD 后,生成一遍 INSTALL 项(在解决方案的CMakeTargets下),此时 build 文件夹内出现了 install 文件夹,这就是所编译的结果,与 openCV 官网的下载包中的 build 对应,对全局变量等的修改也一一对应即可。



【本文地址】


今日新闻


推荐新闻


CopyRight 2018-2019 办公设备维修网 版权所有 豫ICP备15022753号-3